package proyecto.dao;

import java.sql.Connection;
import java.sql.Date;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;
import javax.sql.DataSource;
import proyecto.excepcion.DAOExcepcion;
import proyecto.modelo.Especialidad;

public class EspecialidadDAOImpl extends BaseDAO implements EspecialidadDAO {

    private DataSource dataSource;

    public DataSource getDataSource() {
        return dataSource;
    }

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public Collection<Especialidad> buscarPorCodigo(int codigoEspecialidad)
            throws DAOExcepcion {
        String query = "select codigoEspecialidad, nombre from especialidad where codigoEspecialidad like ?";
        Collection<Especialidad> lista = new ArrayList<Especialidad>();
        Connection con = null;
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            con = dataSource.getConnection();
            stmt = con.prepareStatement(query);
            stmt.setString(1, "%" + codigoEspecialidad + "%");
            rs = stmt.executeQuery();
            while (rs.next()) {
                Especialidad vo = new Especialidad();
                vo.setCodigoEspecialidad(rs.getInt("codigoEspecialidad"));
                vo.setNombre(rs.getString("nombre"));
                lista.add(vo);
            }
        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarResultSet(rs);
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
        System.out.println(lista.size());
        return lista;
    }

    public Especialidad insertar(Especialidad vo) throws DAOExcepcion {
        String query = "INSERT INTO especialidad(nombre) VALUES (?)";
        Connection con = null;
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            con = dataSource.getConnection();
            stmt = con.prepareStatement(query);
            stmt.setString(1, vo.getNombre());
            int i = stmt.executeUpdate();
            if (i != 1) {
                throw new SQLException("No se pudo insertar");
            }
            // Obtener el ultimo id
            int id = 0;
            query = "SELECT LAST_INSERT_ID()";
            stmt = con.prepareStatement(query);
            rs = stmt.executeQuery();
            if (rs.next()) {
                id = rs.getInt(1);
            }
            vo.setCodigoEspecialidad(id);

        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarResultSet(rs);
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
        return vo;
    }

    public Especialidad obtener(int codigoEspecialidad) throws DAOExcepcion {
        Especialidad vo = new Especialidad();
        Connection con = null;
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            String query = "select codigoEspecialidad, nombre from especialidad where codigoEspecialidad=?";
            con = dataSource.getConnection();
            stmt = con.prepareStatement(query);
            stmt.setInt(1, codigoEspecialidad);
            rs = stmt.executeQuery();
            if (rs.next()) {
                vo.setCodigoEspecialidad(rs.getInt("codigoEspecialidad"));
                vo.setNombre(rs.getString("nombre"));
            }
        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarResultSet(rs);
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
        return vo;
    }

    /*
    public void eliminar(int codigoEspecialidad) throws DAOExcepcion {
        String query = "DELETE FROM especialidad WHERE codigoEspecialidad=?";
        Connection con = null;
        PreparedStatement stmt = null;
        try {
            con = dataSource.getConnection();
            stmt = con.prepareStatement(query);
            stmt.setInt(1, codigoEspecialidad);
            int i = stmt.executeUpdate();
            if (i != 1) {
                throw new SQLException("No se pudo eliminar");
            }
        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
    }*/

    public Especialidad actualizar(Especialidad vo) throws DAOExcepcion {
        String query = "update especialidad set nombre=? where codigoEspecialidad=?";
        Connection con = null;
        PreparedStatement stmt = null;
        try {
            con = dataSource.getConnection();
            stmt = con.prepareStatement(query);
            stmt.setString(1, vo.getNombre());
            stmt.setInt(2, vo.getCodigoEspecialidad());
            int i = stmt.executeUpdate();
            if (i != 1) {
                throw new SQLException("No se pudo actualizar");
            }
        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
        return vo;
    }

    public Collection<Especialidad> listar() throws DAOExcepcion {
        Collection<Especialidad> c = new ArrayList<Especialidad>();
        Connection con = null;
        PreparedStatement stmt = null;
        ResultSet rs = null;
        try {
            con = dataSource.getConnection();
            String query = "SELECT codigoEspecialidad, nombre from especialidad";
            stmt = con.prepareStatement(query);
            rs = stmt.executeQuery();
            while (rs.next()) {
                Especialidad vo = new Especialidad();

                vo.setCodigoEspecialidad(rs.getInt("codigoEspecialidad"));
                vo.setNombre(rs.getString("nombre"));

                c.add(vo);
            }

        } catch (SQLException e) {
            System.err.println(e.getMessage());
            throw new DAOExcepcion(e.getMessage());
        } finally {
            this.cerrarResultSet(rs);
            this.cerrarStatement(stmt);
            this.cerrarConexion(con);
        }
        return c;
    }
}
